from setuptools import setup
from torch.utils import cpp_extension

setup(name='my_add',
      ext_modules=[
          cpp_extension.CppExtension('my_ops', ['op.cpp'])
      ],
      cmdclass={'build_ext': cpp_extension.BuildExtension})